multiclass segmentation in tensorflow